msg_tool\scripts\bgi\audio/
audio.rs

1//! Buriko General Interpreter/Ethornell Audio File (Ogg/Vorbis)
2use crate::ext::io::*;
3use crate::scripts::base::*;
4use crate::types::*;
5use anyhow::Result;
6use std::io::{Read, Seek, SeekFrom, Write};
7
8#[derive(Debug)]
9/// Builder for BGI Audio scripts.
10pub struct BgiAudioBuilder {}
11
12impl BgiAudioBuilder {
13    /// Creates a new instance of `BgiAudioBuilder`.
14    pub fn new() -> Self {
15        Self {}
16    }
17}
18
19impl ScriptBuilder for BgiAudioBuilder {
20    fn default_encoding(&self) -> Encoding {
21        Encoding::Utf8
22    }
23
24    fn build_script(
25        &self,
26        buf: Vec<u8>,
27        _filename: &str,
28        _encoding: Encoding,
29        _archive_encoding: Encoding,
30        config: &ExtraConfig,
31        _archive: Option<&Box<dyn Script>>,
32    ) -> Result<Box<dyn Script>> {
33        Ok(Box::new(BgiAudio::new(MemReader::new(buf), config)?))
34    }
35
36    fn build_script_from_file(
37        &self,
38        filename: &str,
39        _encoding: Encoding,
40        _archive_encoding: Encoding,
41        config: &ExtraConfig,
42        _archive: Option<&Box<dyn Script>>,
43    ) -> Result<Box<dyn Script>> {
44        let file = std::fs::File::open(filename)?;
45        let f = std::io::BufReader::new(file);
46        Ok(Box::new(BgiAudio::new(f, config)?))
47    }
48
49    fn build_script_from_reader(
50        &self,
51        reader: Box<dyn ReadSeek>,
52        _filename: &str,
53        _encoding: Encoding,
54        _archive_encoding: Encoding,
55        config: &ExtraConfig,
56        _archive: Option<&Box<dyn Script>>,
57    ) -> Result<Box<dyn Script>> {
58        Ok(Box::new(BgiAudio::new(reader, config)?))
59    }
60
61    fn extensions(&self) -> &'static [&'static str] {
62        &[]
63    }
64
65    fn script_type(&self) -> &'static ScriptType {
66        &ScriptType::BGIAudio
67    }
68
69    fn is_this_format(&self, _filename: &str, buf: &[u8], buf_len: usize) -> Option<u8> {
70        if buf_len >= 8 && buf[4..].starts_with(b"bw  ") {
71            Some(10)
72        } else {
73            None
74        }
75    }
76}
77
78#[derive(Debug)]
79/// BGI Audio script.
80pub struct BgiAudio {
81    data: MemReader,
82}
83
84impl BgiAudio {
85    /// Creates a new instance of `BgiAudio` from a reader.
86    ///
87    /// * `reader` - The reader to read the audio data from.
88    /// * `config` - Extra configuration options.
89    pub fn new<R: Read + Seek>(mut reader: R, _config: &ExtraConfig) -> Result<Self> {
90        let offset = reader.read_u32()?;
91        let len = reader.stream_length()?;
92        if (offset as u64) > len {
93            return Err(anyhow::anyhow!("Invalid offset in BGI audio file"));
94        }
95        let mut magic = [0; 4];
96        reader.read_exact(&mut magic)?;
97        if magic != *b"bw  " {
98            return Err(anyhow::anyhow!(
99                "Invalid magic in BGI audio file: {:?}",
100                magic
101            ));
102        }
103        reader.seek(SeekFrom::Start(offset as u64))?;
104        let mut data = Vec::new();
105        reader.read_to_end(&mut data)?;
106        Ok(Self {
107            data: MemReader::new(data),
108        })
109    }
110}
111
112impl Script for BgiAudio {
113    fn default_output_script_type(&self) -> OutputScriptType {
114        OutputScriptType::Custom
115    }
116
117    fn default_format_type(&self) -> FormatOptions {
118        FormatOptions::None
119    }
120
121    fn is_output_supported(&self, output: OutputScriptType) -> bool {
122        matches!(output, OutputScriptType::Custom)
123    }
124
125    fn custom_output_extension<'a>(&'a self) -> &'a str {
126        "ogg"
127    }
128
129    fn custom_export(&self, filename: &std::path::Path, _encoding: Encoding) -> Result<()> {
130        let mut writer = std::fs::File::create(filename)?;
131        writer.write_all(&self.data.data)?;
132        writer.flush()?;
133        Ok(())
134    }
135}
msg_tool\scripts\bgi\audio/audio.rs

msg_tool\scripts\bgi\audio/
audio.rs
